54154ef4da10ed883834e5f631a3660fed722e23,sniffy-core/src/main/java/io/sniffy/servlet/SnifferServlet.java,SnifferServlet,service,#HttpServletRequest#HttpServletResponse#,50
Before Change
SocketsRegistry.INSTANCE.getDiscoveredAdresses();
if (discoveredAdresses.isEmpty()) {
response.setStatus(HttpServletResponse.SC_OK);
response.setContentType(JAVASCRIPT_MIME_TYPE);
response.flushBuffer();
} else {
After Change
}
} else if (path.equals(SOCKET_REGISTRY_URI_PREFIX)) {
response.setStatus(HttpServletResponse.SC_OK);
response.setContentType(JAVASCRIPT_MIME_TYPE);
Map<Map.Entry<String, Integer>, SocketsRegistry.SocketAddressStatus> discoveredAdresses =
SocketsRegistry.INSTANCE.getDiscoveredAdresses();
if (discoveredAdresses.isEmpty()) {
response.flushBuffer();
} else {
Iterator<Map.Entry<Map.Entry<String, Integer>, SocketsRegistry.SocketAddressStatus>> iterator =
discoveredAdresses.entrySet().iterator();
PrintWriter writer = response.getWriter();
writer.write('[');
while (iterator.hasNext()) {
Map.Entry<Map.Entry<String,Integer>, SocketsRegistry.SocketAddressStatus> entry = iterator.next();
String hostName = entry.getKey().getKey();
Integer port = entry.getKey().getValue();
writer.write('{');
if (null != hostName) {
writer.write("\"host\":\"");
writer.write(hostName);
writer.write("\"");
}
if (null != port) {
if (null != hostName) writer.write(',');
writer.write("\"port\":\"");
writer.write(port.toString());
writer.write("\"");
}
writer.write(',');
writer.write("\"status\":\"");
writer.write(entry.getValue().name());
writer.write("\"");
writer.write('}');
if (iterator.hasNext()) writer.write(',');
}
writer.write(']');
writer.flush();
}
} else if (path.startsWith(SOCKET_REGISTRY_URI_PREFIX)) {
SocketsRegistry.SocketAddressStatus status = null;
if ("POST".equalsIgnoreCase(request.getMethod())) {
status = OPEN;
} else if ("DELETE".equalsIgnoreCase(request.getMethod())) {
status = CLOSED;
}
if (null != status) {
String socketAddress = path.substring(SOCKET_REGISTRY_URI_PREFIX.length());
SocketsRegistry.INSTANCE.setSocketAddressStatus(socketAddress, status);
response.setStatus(HttpServletResponse.SC_CREATED);
response.flushBuffer();
}
}